package com.mmtech.edb.data.util;

import net.sf.jxls.exception.ParsePropertyException;
import net.sf.jxls.transformer.XLSTransformer;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.Workbook;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 导出excel的工具类
 *
 * @author wuchong
 * @date 2018年7月24日 下午5:49:17
 */
public final class ExcelExportUtil {

    private ExcelExportUtil() {
    }

    /**
     * excel导出实现方法
     *
     * @param srcFilePath  模板xls或xlsx文件路径
     * @param objects      模板xls中对应要用到的集合;
     * @param destFilePath 生成的xls或xlsx文件路径;
     * @since 2019年7月13日 下午6:29:14
     */
    public static void createExcel(String srcFilePath, List<?> objects, String destFilePath) {
        /* ********我们也可以使用相对路径来定位 读取模板文件,或放置生成的文件******** */
        // 根据类加载器,获取URL
        // URL url = this.getClass().getClassLoader().getResource("");
        // 获取到项目的classes目录(如:'D:/java/Abc_ExportExcelByTemplate/target/classes/')
        // String srcPath = url.getPath();
        /* ********************************************************* */

        // 创建XLSTransformer对象
        XLSTransformer transformer = new XLSTransformer();
        Map<String, Object> beanParams = new HashMap<>();
        // 将要用到的list集合,按对应模板中的名字,放入map中
        beanParams.put("list", objects);
        try {
            // 生成Excel文件
            transformer.transformXLS(srcFilePath, beanParams, destFilePath);
        } catch (ParsePropertyException | IOException | InvalidFormatException e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取excel文件bytes
     *
     * @param inputStream
     * @param beanParams
     * @return
     * @throws IOException
     * @throws InvalidFormatException
     */
    public static byte[] createExcelBytes(InputStream inputStream, Map<String, Object> beanParams) throws IOException, InvalidFormatException {
        // 创建XLSTransformer对象
        XLSTransformer transformer = new XLSTransformer();
        try (ByteArrayOutputStream swapStream = new ByteArrayOutputStream()) {
            Workbook workbook = transformer.transformXLS(inputStream, beanParams);
            workbook.write(swapStream);
            swapStream.flush();
            return swapStream.toByteArray();
        } finally {
            inputStream.close();
        }
    }


    public static void main(String[] args) {
        System.out.println(~-6);
    }
}